{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [],
   "source": [
    "import chromadb\n",
    "from sentence_transformers import SentenceTransformer\n",
    "from pathlib import Path"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [],
   "source": [
    "client = chromadb.PersistentClient(path=str(Path(\"../chroma_db\").resolve()))\n",
    "\n",
    "embedding_model = SentenceTransformer('all-MiniLM-L6-v2')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [],
   "source": [
    "get_params = {\n",
    "    \"limit\": 10,\n",
    "    \"offset\": 0,\n",
    "    \"include\": [\"metadatas\", \"documents\"]\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "#get_params[\"where_document\"] = {\"$contains\": \"谷春阳\"}\n",
    "get_params[\"where\"] = {\"segment_index\":3}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [],
   "source": [
    "collection = client.get_collection(name=\"document_segments\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "result = collection.get(**get_params)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'ids': ['c852035b-8d73-4948-9d79-55c8e5545f85_3', '60167cb0-6212-4ff5-80ec-dd7545e80e02_3', '87902378-ce49-403d-8e10-8d00daadfb06_3', 'f1ed5afd-172f-4ac8-87e5-744a5f388731_3'], 'embeddings': None, 'documents': ['\\n赵宇明 2025-01-23 08:55:40 \\n行。 \\n赵宇明 2025-01-23 08:55:43 \\n对吧，那个北北京医院弄完没？弄完不就是数据导完没导完，现在就是就是数数据，这个门诊住院的都有没有？ \\n周龙 2025-01-23 08:55:49 \\n那个正在签的接口呢，那个就是。 \\n周龙 2025-01-23 08:55:58 \\n这这些都是正常的 PDF 就是实时的，这不是那个江哥安排那个李鹏飞在在侧着，完了之后江哥在找那个 PDF 的历史数据，他每天在那弄着。 \\n赵宇明 2025-01-23 08:55:59 \\n提不提。PDF 现在起起过来了？ \\n赵宇明 2025-01-23 08:56:17 \\n这就还就还是没完呢，没事。什么时候能还不知道？ \\n周龙 2025-01-23 08:56:19 \\n没有。 \\n周龙 2025-01-23 08:56:22 \\n江哥跟我说的是，他那天说在年前把 PDF 肯定能整完。就是历史的。 \\n赵宇明 2025-01-23 08:56:31 \\n年前没几天了，小心点，咱别大过年的，咱整事到了大过年的别别别别折腾事好吧，对！ \\n周龙 2025-01-23 08:56:37 \\n对。 \\n赵宇明 2025-01-23 08:56:44 \\n行吧，那就就先这样吧！ \\n谷春阳 2025-01-23 08:56:47 \\n行，那我结束会议了。 \\n', '\\n赵宇明 2025-01-23 08:55:40 \\n行。 \\n赵宇明 2025-01-23 08:55:43 \\n对吧，那个北北京医院弄完没？弄完不就是数据导完没导完，现在就是就是数数据，这个门诊住院的都有没有？ \\n周龙 2025-01-23 08:55:49 \\n那个正在签的接口呢，那个就是。 \\n周龙 2025-01-23 08:55:58 \\n这这些都是正常的 PDF 就是实时的，这不是那个江哥安排那个李鹏飞在在侧着，完了之后江哥在找那个 PDF 的历史数据，他每天在那弄着。 \\n赵宇明 2025-01-23 08:55:59 \\n提不提。PDF 现在起起过来了？ \\n赵宇明 2025-01-23 08:56:17 \\n这就还就还是没完呢，没事。什么时候能还不知道？ \\n周龙 2025-01-23 08:56:19 \\n没有。 \\n周龙 2025-01-23 08:56:22 \\n江哥跟我说的是，他那天说在年前把 PDF 肯定能整完。就是历史的。 \\n赵宇明 2025-01-23 08:56:31 \\n年前没几天了，小心点，咱别大过年的，咱整事到了大过年的别别别别折腾事好吧，对！ \\n周龙 2025-01-23 08:56:37 \\n对。 \\n赵宇明 2025-01-23 08:56:44 \\n行吧，那就就先这样吧！ \\n谷春阳 2025-01-23 08:56:47 \\n行，那我结束会议了。 \\n', ' 的后端开发。 \\n谷春阳 2025-01-23 08:51:24 \\n开始写一下代码。我这边完了，下一个，下一个。 \\n谷春阳 2025-01-23 08:51:35 \\n马云松。 \\n马雪松 2025-01-23 08:51:36 \\n我昨天的话是做的高淳的那个渗透的采集的报数，然后还有就是把这个数据采集了，然后今天的话首先先看高速那边有没有，就是他的系统准备好了，准备好了，然后做那个。 \\n马雪松 202', '24 \\n开始写一下代码。我这边完了，下一个，下一个。 \\n谷春阳 2025-01-23 08:51:35 \\n马云松。 \\n马雪松 2025-01-23 08:51:36 \\n我昨天的话是做的高淳的那个渗透的采集的报数，然后还有就是把这个数据采集了，然后今天的话首先先看高速那边有没有，就是他的系统准备好了，准备好了，然后做那个。 \\n马雪松 2025-01-23 08:51:58 \\n其他系统的采集如果没有的'], 'uris': None, 'data': None, 'metadatas': [{'document_id': 'c852035b-8d73-4948-9d79-55c8e5545f85', 'segment_index': 3}, {'document_id': '60167cb0-6212-4ff5-80ec-dd7545e80e02', 'segment_index': 3}, {'document_id': '87902378-ce49-403d-8e10-8d00daadfb06', 'segment_index': 3}, {'document_id': 'f1ed5afd-172f-4ac8-87e5-744a5f388731', 'segment_index': 3}], 'included': [<IncludeEnum.documents: 'documents'>, <IncludeEnum.metadatas: 'metadatas'>]}\n"
     ]
    }
   ],
   "source": [
    "print(result)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "langflow",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
